image classification using RESNET model in pytorch